package work;

import java.util.HashMap;
import java.util.Map;

/**
 * 和为K的子数组
 * 时间复杂度：O(n）
 * 思路：1）求前缀和；2）计数
 */
public class SubarraySum {
    public int subarraySum(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>(); //计数
        map.put(0, 1); //注意点
        int res = 0; //结果
        int presum = 0; //nums的前缀和
        for(int i = 0; i < nums.length; i++){
            presum += nums[i];
            if(map.containsKey(presum - k)){ //注意点：判断要放在添加前面，否则当k=0时，就会把自身加一遍
                res += map.get(presum - k);
            }
            if(map.containsKey(presum)){
                map.put(presum, map.get(presum) + 1);
            }else{
                map.put(presum, 1);
            }
        }
        return res;
    }
}
